How to wrap __getitem__ method correctly to keep square brackets functionality